import os
from pydantic import BaseModel
from pathlib import Path

class Config(BaseModel):

    # 应用
    APP_NAME: str = "IPO2"
    # 大模型api_key
    API_KEY: str = ""
    # 大模型名称
    MODEL: str = "qwen3-235b-a22b"

    # 项目名
    PROJECT_DIR: str = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

    # 保存处理过的中间文件
    DATA_DIR: str = os.path.join(PROJECT_DIR, 'data')

    # 保存模型
    MODEL_DIR: str = os.path.join(PROJECT_DIR, 'model')

    # vector
    VECTOR_DIR: str = os.path.join(PROJECT_DIR, 'vector')

config = Config()


if __name__ == '__main__':
    print(config.PROJECT_DIR)